透過上篇的創建模組,我們得到了許多的檔案,接下來我們需要透過檢查點的變數,將他轉換成常數節點,同時讓這些節點與圖像結合成獨立的檔案,這樣之後可以再APP端可以更簡單的使用這個模組檔
import tensorflow as tf
from tensorflow.python.tools import freeze_graph
input_graph = 'android_tf.pbtxt'
input_saver = ""
input_binary = False
checkpoint = 'android_tf.ckpt'
output_name = "O"
restore_name = "save/restore_all"
tensor_name = "save/Const:0"
output_frozen_graph = 'frozen_android_tf.pb'
clear_devices = True
freeze_graph.freeze_graph(input_graph,input_saver,
input_binary,checkpoint,output_name,
restore_name,tensor_name,
output_frozen_graph,clear_devices,"")